#ifndef CONE_DETECTOR_H_
#define CONE_DETECTOR_H_

#include "rein/algorithms/detection/detector.h"

namespace rein {

class ConeDetector : public Detector
{
 public:
  ConeDetector(ModelStoragePtr model_storage) : Detector(model_storage) {};

  /**
  * \brief Run the object detector.
  */
  virtual void detect();

  /**
  * Each detector will have an unique name. This returns the detector name.
  * @return name of the detector
  */
  virtual std::string getName() { return "Cone"; }

  /**
  * Loads pre-trained models for a list of objects.
  * @param models list of objects to load
  */
  virtual void loadModels(const std::vector<std::string>& models);
};

}
#endif /* CONE_DETECTOR_H_ */
